package handler

import (
	"github.com/gin-gonic/gin"
	"zk3zc/dao"
	"zk3zc/pkg"
	"zk3zc/request"
)

func Upload(c *gin.Context) {
	file, err := c.FormFile("file")
	if err != nil {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}

	ext := file.Filename

	if ext != ".jpg" && ext != ".png" && ext != ".mp4" {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  "文件格式不对",
			"data": nil,
		})
		return
	}
	if file.Size > 500*1024*1024 {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  "文件大小只能在500M以内",
			"data": nil,
		})
		return
	}

	// c.JSON(200, gin.H{"message": file.Header.Context})
	c.SaveUploadedFile(file, file.Filename)

	lock := "C:\\Users\\Helen\\GoWork\\2209A\\专高三\\zy16\\zk3zc\\img\\" + file.Filename
	c.JSON(200, gin.H{
		"code": 0,
		"msg":  "成功",
		"data": lock,
	})

}

func AddArt(c *gin.Context) {
	var data request.AddArt

	if err := c.ShouldBind(&data); err != nil {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	userId := c.GetUint("userId")
	err := dao.AddArt(data.Title, data.Context, data.Level, int(userId))
	if err != nil {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	c.JSON(200, gin.H{
		"code": 0,
		"msg":  "成功",
		"data": nil,
	})

}

func FockArt(c *gin.Context) {
	var data request.Fock

	if err := c.ShouldBind(&data); err != nil {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
}

func Login(c *gin.Context) {
	var data request.Login

	if err := c.ShouldBind(&data); err != nil {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	login, err := dao.Login(data.Mobile, data.Password)
	if err != nil {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	if login.ID == 0 {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  "用户不存在",
			"data": nil,
		})
		return
	}
	u := pkg.CustomClaims{
		ID: login.ID,
	}
	token, err := pkg.NewJWT("2209a").CreateToken(u)
	if err != nil {
		c.JSON(200, gin.H{
			"code": 1000,
			"msg":  err.Error(),
			"data": nil,
		})
		return
	}
	c.JSON(200, gin.H{
		"code": 0,
		"msg":  "成功",
		"data": token,
	})
}
